# https://github.com/FlagOpen/FlagEmbedding/tree/master/examples/finetune/reranker
torchrun --nproc_per_node 1 \
-m FlagEmbedding.reranker.run \
--output_dir /home/liwei/bge-reranker-v2-m3/sishu \
--model_name_or_path  /home/liwei/bge-reranker-v2-m3/ \
--train_data ./data/sishu/sishu_bge_train.jsonl \
--learning_rate 6e-5 \
--fp16 \
--num_train_epochs 5 \
--per_device_train_batch_size 1 \
--gradient_accumulation_steps 4 \
--dataloader_drop_last True \
--train_group_size 16 \
--max_len 256 \
--weight_decay 0.01 \
--logging_steps 10